import { setWorkerDrain } from "@autoscholar/core";
import { z } from "zod";
import { handler, json, parseBody } from "@/lib/server/api";
import { db } from "@/lib/server/db";
import { requireAdmin } from "@/lib/server/session";

export const runtime = "nodejs";

export const POST = handler(async (req, { params }) => {
  const sess = await requireAdmin(req, { mutation: true });
  const { id } = (await params) as { id: string };
  const body = await parseBody(req, z.object({ drain: z.boolean() }));
  const w = await setWorkerDrain(db(), id, body.drain, { id: sess.userId });
  return json({ worker: { ...w, tokenHash: undefined } });
});